home *** CD-ROM | disk | FTP | other *** search
/ Programming an RTS Game with Direct3D / Programming an RTS Game with Direct3D.iso / Examples / Chapter 12 / Example 12.7 / mesh.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2006-07-01  |  5.2 KB  |  212 lines
  1. #include "mesh.h"
  2.  
  3. const DWORD ObjectVertex::FVF = D3DFVF_XYZ | D3DFVF_NORMAL | D3DFVF_TEX1;
  4.  
  5. MESH::MESH()
  6. {
  7.     m_pDevice = NULL;
  8.     m_pMesh = NULL;
  9. }
  10.  
  11. MESH::MESH(char fName[], IDirect3DDevice9* Dev)
  12. {
  13.     m_pDevice = Dev;
  14.     m_pMesh = NULL;
  15.     Load(fName, m_pDevice);
  16. }
  17.  
  18. MESH::~MESH()
  19. {
  20.     Release();
  21. }
  22.  
  23. HRESULT MESH::Load(char fName[], IDirect3DDevice9* Dev)
  24. {
  25.     m_pDevice = Dev;
  26.  
  27.     //Set white material
  28.     m_white.Ambient = m_white.Specular = m_white.Diffuse  = D3DXCOLOR(1.0f, 1.0f, 1.0f, 1.0f);
  29.     m_white.Emissive = D3DXCOLOR(0.0f, 0.0f, 0.0f, 1.0f);
  30.     m_white.Power = 1.0f;
  31.  
  32.     Release();
  33.  
  34.     //Load new mesh
  35.     ID3DXBuffer * adjacencyBfr = NULL;
  36.     ID3DXBuffer * materialBfr = NULL;
  37.     DWORD noMaterials = NULL;
  38.  
  39.     if(FAILED(D3DXLoadMeshFromX(fName, D3DXMESH_MANAGED, m_pDevice, &adjacencyBfr, &materialBfr, NULL, &noMaterials, &m_pMesh)))
  40.         return E_FAIL;
  41.  
  42.     D3DXMATERIAL *mtrls = (D3DXMATERIAL*)materialBfr->GetBufferPointer();
  43.  
  44.     for(int i=0;i<noMaterials;i++)
  45.     {
  46.         m_materials.push_back(mtrls[i].MatD3D);
  47.  
  48.         if(mtrls[i].pTextureFilename != NULL)
  49.         {
  50.             char textureFileName[90];
  51.             strcpy(textureFileName, "meshes/");
  52.             strcat(textureFileName, mtrls[i].pTextureFilename);
  53.             IDirect3DTexture9 * newTexture = NULL;
  54.             D3DXCreateTextureFromFile(m_pDevice, textureFileName, &newTexture);            
  55.             m_textures.push_back(newTexture);
  56.         }
  57.         else m_textures.push_back(NULL);
  58.     }
  59.  
  60.     m_pMesh->OptimizeInplace(D3DXMESHOPT_ATTRSORT | D3DXMESHOPT_COMPACT | D3DXMESHOPT_VERTEXCACHE,
  61.                             (DWORD*)adjacencyBfr->GetBufferPointer(), NULL, NULL, NULL);
  62.  
  63.     adjacencyBfr->Release();
  64.     materialBfr->Release();
  65.  
  66.     return S_OK;
  67. }
  68.  
  69. void MESH::Render()
  70. {
  71.     for(int i=0;i<m_materials.size();i++)
  72.     {    
  73.         if(m_textures[i] != NULL)m_pDevice->SetMaterial(&m_white);
  74.         else m_pDevice->SetMaterial(&m_materials[i]);
  75.         m_pDevice->SetTexture(0,m_textures[i]);
  76.         m_pMesh->DrawSubset(i);
  77.     }    
  78. }
  79.  
  80. void MESH::Render(float prc)
  81. {
  82.     //Calculate the end subset to render
  83.     int end = m_materials.size() * prc;
  84.  
  85.     for(int i=0;i<m_materials.size() && i < end;i++)
  86.     {        
  87.         if(m_textures[i] != NULL)m_pDevice->SetMaterial(&m_white);
  88.         else m_pDevice->SetMaterial(&m_materials[i]);
  89.         m_pDevice->SetTexture(0,m_textures[i]);
  90.         m_pMesh->DrawSubset(i);
  91.     }
  92. }
  93.  
  94. void MESH::Release()
  95. {
  96.     //Clear old mesh...
  97.     if(m_pMesh != NULL)
  98.     {
  99.         m_pMesh->Release();
  100.         m_pMesh = NULL;
  101.     }
  102.  
  103.     //Clear textures and materials
  104.     for(int i=0;i<m_textures.size();i++)
  105.         if(m_textures[i] != NULL)
  106.             m_textures[i]->Release();
  107.  
  108.     m_textures.clear();
  109.     m_materials.clear();    
  110. }
  111.  
  112. MESHINSTANCE::MESHINSTANCE()
  113. {
  114.     m_pMesh = NULL;
  115.     m_pos = m_rot = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  116.     m_sca = D3DXVECTOR3(1.0f, 1.0f, 1.0f);
  117. }
  118.  
  119. MESHINSTANCE::MESHINSTANCE(MESH *meshPtr)
  120. {
  121.     m_pMesh = meshPtr;
  122.     m_pos = m_rot = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  123.     m_sca = D3DXVECTOR3(1.0f, 1.0f, 1.0f);
  124. }
  125.  
  126. D3DXMATRIX MESHINSTANCE::GetWorldMatrix()
  127. {
  128.     D3DXMATRIX p, r, s;
  129.     D3DXMatrixTranslation(&p, m_pos.x, m_pos.y, m_pos.z);
  130.     D3DXMatrixRotationYawPitchRoll(&r, m_rot.y, m_rot.x, m_rot.z);
  131.     D3DXMatrixScaling(&s, m_sca.x, m_sca.y, m_sca.z);
  132.  
  133.     D3DXMATRIX world = s * r * p;
  134.     return world;
  135. }
  136.  
  137. void MESHINSTANCE::Render()
  138. {
  139.     if(m_pMesh != NULL)
  140.     {
  141.         m_pMesh->m_pDevice->SetTransform(D3DTS_WORLD, &GetWorldMatrix());
  142.         m_pMesh->Render();
  143.     }
  144. }
  145.  
  146. BBOX MESHINSTANCE::GetBoundingBox()
  147. {
  148.     if(m_pMesh == NULL || m_pMesh->m_pMesh == NULL)return BBOX();
  149.     
  150.     if(m_pMesh->m_pMesh->GetFVF() != ObjectVertex::FVF)        // XYZ and NORMAL and UV
  151.         return BBOX();
  152.  
  153.     BBOX bBox(D3DXVECTOR3(-10000.0f, -10000.0f, -10000.0f),
  154.               D3DXVECTOR3(10000.0f, 10000.0f, 10000.0f));
  155.     D3DXMATRIX World = GetWorldMatrix();
  156.  
  157.     //Lock vertex buffer of the object
  158.     ObjectVertex* vertexBuffer = NULL;
  159.     m_pMesh->m_pMesh->LockVertexBuffer(0,(void**)&vertexBuffer);
  160.  
  161.     //For each vertex in the mesh
  162.     for(int i=0;i<m_pMesh->m_pMesh->GetNumVertices();i++)
  163.     {
  164.         //Transform vertex to world space using the MESHINSTANCE
  165.         //world matrix, i.e. the position, rotation and scale
  166.         D3DXVECTOR3 pos;
  167.         D3DXVec3TransformCoord(&pos, &vertexBuffer[i]._pos, &World);
  168.  
  169.         // Check if the vertex is outside the bounds
  170.         // if so, then update the bounding volume
  171.         if(pos.x < bBox.min.x)bBox.min.x = pos.x;
  172.         if(pos.x > bBox.max.x)bBox.max.x = pos.x;
  173.         if(pos.y < bBox.min.y)bBox.min.y = pos.y;
  174.         if(pos.y > bBox.max.y)bBox.max.y = pos.y;
  175.         if(pos.z < bBox.min.z)bBox.min.z = pos.z;
  176.         if(pos.z > bBox.max.z)bBox.max.z = pos.z;
  177.     }
  178.  
  179.     m_pMesh->m_pMesh->UnlockVertexBuffer();
  180.  
  181.     return bBox;
  182. }
  183.  
  184. BSPHERE MESHINSTANCE::GetBoundingSphere()
  185. {
  186.     if(m_pMesh == NULL || m_pMesh->m_pMesh == NULL)return BSPHERE();
  187.     if(m_pMesh->m_pMesh->GetFVF() != ObjectVertex::FVF)        // XYZ and NORMAL and UV
  188.         return BSPHERE();
  189.  
  190.     BBOX bBox = GetBoundingBox();
  191.     BSPHERE bSphere;
  192.     D3DXMATRIX World = GetWorldMatrix();
  193.     bSphere.center = (bBox.max + bBox.min) / 2.0f;
  194.  
  195.     ObjectVertex* vertexBuffer = NULL;
  196.     m_pMesh->m_pMesh->LockVertexBuffer(0,(void**)&vertexBuffer);
  197.  
  198.     //Get radius
  199.     for(int i=0;i<m_pMesh->m_pMesh->GetNumVertices();i++)
  200.     {
  201.         D3DXVECTOR3 pos;
  202.         D3DXVec3TransformCoord(&pos, &vertexBuffer[i]._pos, &World);
  203.  
  204.         float l = D3DXVec3Length(&(pos - bSphere.center));
  205.         if(l > bSphere.radius)
  206.             bSphere.radius = l;
  207.     }
  208.  
  209.     m_pMesh->m_pMesh->UnlockVertexBuffer();
  210.  
  211.     return bSphere;
  212. }